package com.zhouyihe.weblog.common.exception;

import com.zhouyihe.weblog.common.enums.ResponseCodeEnum;
import com.zhouyihe.weblog.common.utils.Response;
import lombok.extern.slf4j.Slf4j;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.servlet.http.HttpServletRequest;
import java.util.Optional;

/**
 * @author 周益和
 * @ClassName GlobalExceptionHandler
 * @qq 1552951165
 * @date 2024/5/26
 */
@ControllerAdvice
@Slf4j
public class GlobalExceptionHandler {
    
    /**
     * 捕获参数校验异常
     * @return
     */
    @ExceptionHandler({ MethodArgumentNotValidException.class })
    @ResponseBody
    public Response<Object> handleMethodArgumentNotValidException(HttpServletRequest request
            , MethodArgumentNotValidException e) {
        // 参数错误异常码
        String errorCode = ResponseCodeEnum.PARAM_NOT_VALID.getErrorCode();
        
        // 获取 BindingResult
        BindingResult bindingResult = e.getBindingResult();
        
        StringBuilder sb = new StringBuilder();
        
        // 获取校验不通过的字段，并组合错误信息，格式为： email 邮箱格式不正确, 当前值: '123124qq.com';
        
        /* 使用Optional包装getFieldErrors()方法的返回值，以防止在该方法返回null时引发NullPointerException */
        /*  对于每个FieldError，将以下信息拼接成一个字符串并添加StringBuilder sb中：
            错误字段的名称（error.getField()）
            默认的错误消息（error.getDefaultMessage()）
            被拒绝的输入值（error.getRejectedValue() */
        Optional.ofNullable(bindingResult.getFieldErrors()).ifPresent(errors -> {
            errors.forEach(error ->
                    sb.append(error.getField())
                            .append(" ")
                            .append(error.getDefaultMessage())
                            .append(", 当前值: '")
                            .append(error.getRejectedValue())
                            .append("'; ")
            
            );
        });
        
        // 错误信息
        String errorMessage = sb.toString();
        
        log.warn("{} request error, errorCode: {}, errorMessage: {}", request.getRequestURI(), errorCode, errorMessage);
        
        return Response.fail(errorCode, errorMessage);
    }
    
    /**
     * 捕获自定义业务异常
     * @return
     */
    @ExceptionHandler({BizException.class})//当发生自定义的BizException异常时，该方法会被调用。
    @ResponseBody
    public Response<Object> handleBizException(HttpServletRequest request, BizException e) {
        log.warn("{} request fail, errorCode: {}, errorMessage: {}", request.getRequestURI(),
                e.getErrorCode(), e.getErrorMessage());
        return Response.fail(e);
    }
    
    /**
     * 其他类型异常
     * @param request
     * @param e
     * @return
     */
    @ExceptionHandler({ Exception.class })//当发生其他的异常时，该方法会被调用。
    @ResponseBody
    public Response<Object> handleOtherException(HttpServletRequest request, Exception e) {
        log.error("{} request error, ", request.getRequestURI(), e);
        return Response.fail(ResponseCodeEnum.SYSTEM_ERROR);
    }
    
    @ExceptionHandler({ AccessDeniedException.class })
    public void throwAccessDeniedException(AccessDeniedException e) throws AccessDeniedException {
        // 捕获到鉴权失败异常，主动抛出，交给 RestAccessDeniedHandler 去处理
        log.info("============= 捕获到 AccessDeniedException");
        throw e;
    }
}
